/*
 * To change this template, choose Tools | Templates
 * and open the template in the editor.
 */
package simchewie;

import java.awt.Graphics;
import java.awt.Graphics2D;
import java.awt.GraphicsConfiguration;
import java.awt.GraphicsDevice;
import java.awt.GraphicsEnvironment;
import java.awt.HeadlessException;
import java.awt.Image;
import java.awt.Transparency;
import java.awt.geom.AffineTransform;
import java.awt.geom.Point2D;
import java.awt.image.BufferedImage;
import java.awt.image.PixelGrabber;
import javax.swing.ImageIcon;

/**
 *
 * @author willi
 */
public class Pictures {

    public static BufferedImage toBufferedImage(Image image) {
        if (image instanceof BufferedImage) {
            return (BufferedImage) image;
        }

        // This code ensures that all the pixels in the image are loaded
        image = new ImageIcon(image).getImage();

        // Determine if the image has transparent pixels
        boolean hasAlpha = hasAlpha(image);

        // Create a buffered image with a format that's compatible with the screen
        BufferedImage bimage = null;
        GraphicsEnvironment ge = GraphicsEnvironment.getLocalGraphicsEnvironment();
        try {
            // Determine the type of transparency of the new buffered image
            int transparency = Transparency.OPAQUE;
            if (hasAlpha == true) {
                transparency = Transparency.BITMASK;
            }

            // Create the buffered image
            GraphicsDevice gs = ge.getDefaultScreenDevice();
            GraphicsConfiguration gc = gs.getDefaultConfiguration();
            bimage = gc.createCompatibleImage(image.getWidth(null), image.getHeight(null), transparency);
        } catch (HeadlessException e) {
        } //No screen

        if (bimage == null) {
            // Create a buffered image using the default color model
            int type = BufferedImage.TYPE_INT_RGB;
            if (hasAlpha == true) {
                type = BufferedImage.TYPE_INT_ARGB;
            }
            bimage = new BufferedImage(image.getWidth(null), image.getHeight(null), type);
        }

        // Copy image to buffered image
        Graphics g = bimage.createGraphics();

        // Paint the image onto the buffered image
        g.drawImage(image, 0, 0, null);
        g.dispose();

        return bimage;
    }

    public static boolean hasAlpha(Image image) {
        // If buffered image, the color model is readily available
        if (image instanceof BufferedImage) {
            return ((BufferedImage) image).getColorModel().hasAlpha();
        }

        // Use a pixel grabber to retrieve the image's color model;
        // grabbing a single pixel is usually sufficient
        PixelGrabber pg = new PixelGrabber(image, 0, 0, 1, 1, false);
        try {
            pg.grabPixels();
        } catch (InterruptedException e) {
        }

        // Get the image's color model
        return pg.getColorModel().hasAlpha();
    }

    public static BufferedImage rotateImage(BufferedImage inputImage, int RotAngle) {

        int X, Y = 0;

        if ((RotAngle == 0) || (RotAngle == 180) || (RotAngle == 360)) {
            X = inputImage.getWidth(null);
            Y = inputImage.getHeight(null);
        } else {
            X = inputImage.getHeight(null);
            Y = inputImage.getWidth(null);
        }

        BufferedImage sourceBI = new BufferedImage(X, Y, BufferedImage.TYPE_INT_RGB);

        AffineTransform at = new AffineTransform();

        // rotate around image center
        at.rotate(Math.toRadians(RotAngle), (sourceBI.getWidth() / 2D), (sourceBI.getHeight() / 2D));

        /*
         * translate to make sure the rotation doesn't cut off any image data
         */
        AffineTransform translationTransform;
        translationTransform = findTranslation(at, sourceBI, RotAngle);
        at.preConcatenate(translationTransform);


        Graphics2D g = (Graphics2D) sourceBI.getGraphics();
        g.setTransform(at);
        g.drawImage(inputImage, 0, 0, null);


        return sourceBI;
    }

    private static AffineTransform findTranslation(AffineTransform at, BufferedImage bi, int RotAngle) {
        Point2D p2din, p2dout;
        double ytrans, xtrans = 0.0;

        AffineTransform tat = new AffineTransform();

        if (RotAngle == 180) {
            p2din = new Point2D.Double(0, bi.getHeight());
        } else {
            p2din = new Point2D.Double(0.0, 0.0);
        }

        p2dout = at.transform(p2din, null);

        if (RotAngle == 270) {
            xtrans = p2dout.getX();
            ytrans = xtrans;
        } else {
            ytrans = p2dout.getY();
            xtrans = ytrans;
        }

        //System.out.println("X: "+ (xtrans) +" Y: "+ (ytrans));

        tat.translate(-xtrans, -ytrans);

        return tat;
    }
}
